package com.sheng.leetcode.year2023.month08.day24;

import org.junit.Test;

/**
 * @author by ls
 * @date 2023/8/24
 * <p>
 * 1267. 统计参与通信的服务器<p>
 * <p>
 * 这里有一幅服务器分布图，服务器的位置标识在 m * n 的整数矩阵网格 grid 中，1 表示单元格上有服务器，0 表示没有。<p>
 * 如果两台服务器位于同一行或者同一列，我们就认为它们之间可以进行通信。<p>
 * 请你统计并返回能够与至少一台其他服务器进行通信的服务器的数量。<p>
 * <p>
 * 示例 1：<p>
 * 输入：grid = [[1,0],[0,1]]<p>
 * 输出：0<p>
 * 解释：没有一台服务器能与其他服务器进行通信。<p>
 * <p>
 * 示例 2：<p>
 * 输入：grid = [[1,0],[1,1]]<p>
 * 输出：3<p>
 * 解释：所有这些服务器都至少可以与一台别的服务器进行通信。<p>
 * <p>
 * 示例 3：<p>
 * 输入：grid = [[1,1,0,0],[0,0,1,0],[0,0,1,0],[0,0,0,1]]<p>
 * 输出：4<p>
 * 解释：第一行的两台服务器互相通信，第三列的两台服务器互相通信，但右下角的服务器无法与其他服务器通信。<p>
 * <p>
 * 提示：<p>
 * m == grid.length<p>
 * n == grid[i].length<p>
 * 1 <= m <= 250<p>
 * 1 <= n <= 250<p>
 * grid[i][j] == 0 or 1<p>
 */
public class LeetCode1267 {

    @Test
    public void test01() {
        int[][] grid = {{1, 0}, {0, 1}};
//        int[][] grid = {{1, 0}, {1, 1}};
//        int[][] grid = {{1, 1, 0, 0}, {0, 0, 1, 0}, {0, 0, 1, 0}, {0, 0, 0, 1}};
        System.out.println(new Solution().countServers(grid));
    }
}

class Solution {
    public int countServers(int[][] grid) {
        // 1 表示单元格上有服务器，0 表示没有
        // 可以统计每行中 1 的数量，大于等于 2 时，直接加上结果，但是每列中的 1，也要计算在内
        int m = grid.length;
        int n = grid[0].length;
        int ans = 0;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                // 暴力统计同行和同列中是否存在 1
                if (grid[i][j] == 1) {
                    boolean flag = false;
                    for (int k = 0; k < m; k++) {
                        if (i != k && grid[k][j] == 1) {
                            flag = true;
                            break;
                        }
                    }
                    if (!flag) {
                        for (int k = 0; k < n; k++) {
                            if (j != k && grid[i][k] == 1) {
                                flag = true;
                                break;
                            }
                        }
                    }
                    if (flag) {
                        ans++;
                    }
                }
            }
        }
        return ans;
    }
}
